from easycore.common.registry import Registry

ARCH_REGISTRY = Registry('arch')

def build_model(cfg):
    model = ARCH_REGISTRY.get(cfg.MODEL.ARCHITECTURE)(cfg)
    return model